package framework

import (
	"github.com/golang-module/carbon"
	"go.uber.org/zap"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"
	"time"
)

var DB *gorm.DB

func init() {
	var dsn string
	dsn = "root:root@tcp(127.0.0.1:3306)/gadmin?charset=utf8mb4&parseTime=True&loc=Local"
	mysqlConfig := mysql.Config{
		DSN:                       dsn,   // DSN data source name
		DefaultStringSize:         256,   // string 类型字段的默认长度
		DisableDatetimePrecision:  false, // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
		DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
		DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
		SkipInitializeWithVersion: true,  // 根据当前 MySQL 版本自动配置
	}
	gormConfig := &gorm.Config{
		Logger: logger.Default.LogMode(logger.Info),
		NamingStrategy: schema.NamingStrategy{
			TablePrefix:   "t_", // 表名前缀，`User`表为`t_users`
			SingularTable: true, // 使用单数表名，启用该选项后，`User` 表将是`user`
		},
		NowFunc: func() time.Time {
			return time.Now().Local()
		},
		DisableForeignKeyConstraintWhenMigrating: true,
		// 添加cache
		PrepareStmt: true,
	}
	db, err := gorm.Open(mysql.New(mysqlConfig), gormConfig)
	if err != nil {
		Log.Error("数据库初始化失败", zap.Error(err))
		return
	}
	DB = db
}

func AutoMigrate(option string, dst interface{}) {
	if err := DB.Set("gorm:table_options", option).AutoMigrate(dst); err != nil {
		Log.Error("表创建失败!", zap.Any("model", dst))
		return
	}
}

type Model struct {
	ID        uint            `json:"id" gorm:"primaryKey;comment:主键;"`
	CreatedAt carbon.DateTime `gorm:"comment:创建时间;"`
	UpdatedAt carbon.DateTime `gorm:"comment:修改时间;"`
}
